# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     config
   Description :   
   Author :       lth
   date：          2022/1/24
-------------------------------------------------
   Change Activity:
                   2022/1/24 12:28: 
-------------------------------------------------
"""
__author__ = 'lth'


class GetConfig:
    project_name = "SRGAN 超分"
    use_cuda = True
    seed = 123
    resume = True
    pretrained_generator_weight = "./weight/generator_best.pth"
    pretrained_discriminator_weight = "./weight/discriminator_best.pth"
    lr = 0.0001
    milestone = [5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80]
    epochs = 200

    train_batch_size = 10
    test_batch_size = 10
    # base_dir="/media/luotianhang/B24C6EE14C6E9FB9/Datasets/超分"
    base_dir = "E:/Datasets/超分/mirflickr"
    val_per = 0.01

    factor = 4
    lower_res = 96
    higher_res = lower_res * factor
